package com.xsherl.leetcode.solution;

import cn.hutool.core.util.ArrayUtil;
import com.xsherl.leetcode.utils.ArrayUtils;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

/**
 * 给定一个整数数组 nums和一个整数目标值 target，请你在该数组中找出 和为目标值 的那两个整数，并返回它们的数组下标。
 *
 * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
 *
 * 你可以按任意顺序返回答案。
 */
public class TwoSum {

    /**
     * 初版使用HashMap缓存另一个数的下标，如果发现则返回
     */
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i  = 0; i < nums.length; ++i){
            int n = nums[i];
            Integer m = map.get(n);
            if (m != null){
                return new int[]{m, i};
            }
            map.put(target - n, i);
        }
        return new int[0];
    }

    public static void main(String[] args) {
        int[] nums = ArrayUtil.unWrap(ArrayUtils.parseArray("[2,7,11,15]", Integer.class));
        int target = 9;
        int[] ints = new TwoSum().twoSum(nums, target);
        System.out.println(Arrays.toString(ints));
    }

}
